package cn.edu.cqu.bysj.tool.certificate;

import org.apache.commons.httpclient.protocol.SecureProtocolSocketFactory;

import javax.net.ssl.SSLContext;
import javax.net.ssl.TrustManager;
import javax.net.ssl.X509TrustManager;
import java.io.IOException;
import java.net.InetAddress;
import java.net.Socket;
import java.security.KeyManagementException;
import java.security.NoSuchAlgorithmException;
import java.security.cert.CertificateException;
import java.security.cert.X509Certificate;


public class MySecureProtocolSocketFactory implements SecureProtocolSocketFactory {

    private SSLContext sslcontext = null;

    private SSLContext createSSLContext(){
        SSLContext sslcontext = null;
        try{
            sslcontext = SSLContext.getInstance("SSL");
            sslcontext.init(null, new TrustManager[]{
                    new TrustAnyTrustManager()
            }, new java.security.SecureRandom());
        }catch(NoSuchAlgorithmException | KeyManagementException e){
            e.printStackTrace();
        }
        return sslcontext;
    }

    private SSLContext getSSLContext(){
        if(this.sslcontext == null){
            this.sslcontext = createSSLContext();
        }
        return this.sslcontext;
    }

    public Socket createSocket(Socket socket, String host, int port, boolean autoClose) throws IOException {
        return getSSLContext().getSocketFactory().createSocket(socket, host, port, autoClose);
    }

    public Socket createSocket(String host, int port) throws IOException {
        return getSSLContext().getSocketFactory().createSocket(host, port);
    }

    public Socket createSocket(String host, int port, InetAddress clientHost, int clientPort) throws IOException {
        return getSSLContext().getSocketFactory().createSocket(host, port, clientHost, clientPort);
    }

//    public Socket createSocket(String host, int port, InetAddress localAddress, int localPort, HttpConnectionParams params)
//            throws IOException {
//        if(params == null){
//            throw new IllegalArgumentException("Parameters may not be null");
//        }
//        int timeout = 3000;
//        SocketFactory socketfactory = getSSLContext().getSocketFactory();
//        if(timeout == 0){
//            return socketfactory.createSocket(host, port, localAddress, localPort);
//        }else{
//            Socket socket = socketfactory.createSocket();
//            SocketAddress localaddr = new InetSocketAddress(localAddress, localPort);
//            SocketAddress remoteaddr = new InetSocketAddress(host, port);
//            socket.bind(localaddr);
//            socket.connect(remoteaddr, timeout);
//            return socket;
//        }
//    }

    // 自定义私有类
    private static class TrustAnyTrustManager implements X509TrustManager {

        public void checkClientTrusted(X509Certificate[] chain, String authType) throws CertificateException{
        }

        public void checkServerTrusted(X509Certificate[] chain, String authType) throws CertificateException{
        }

        public X509Certificate[] getAcceptedIssuers(){
            return new X509Certificate[]{};
        }
    }
}